import datetime

import litellm
from google.adk import Agent
import os
from google.adk.models.lite_llm import LiteLlm
os.environ['OPENAI_API_BASE'] = 'https://dashscope.aliyuncs.com/compatible-mode/v1'
os.environ["OPENAI_API_KEY"] = "sk-965dc39b016c49ecbe29de180f4db2b6"



def get_weather(city:str)->str:
    return "weather"
litellm.register_model({
"model_name":"doubao/deepseek-v3-250324",
    "api_base":"https://ark.cn-beijing.volces.com/api/v3",
    "api_key":"9d4b99d0-d0a4-4212-9e4d-0ed00c0d313a"
}
    )
def get_current_time():
    return str(datetime.now())

os.environ['OPENAI_API_BASE'] = 'https://ark.cn-beijing.volces.com/api/v3'
os.environ["OPENAI_API_KEY"] = "9d4b99d0-d0a4-4212-9e4d-0ed00c0d313a"
root_agent = Agent(
    model=LiteLlm(model="deepseek-v3-250324"),
    name='root_agent', description="你是一个优秀的AI Agent",
    instruction="请开始你的表演！",
)